import pickle

from torch.utils.data import Dataset
from Gpt2_chatbot.parameter_config import ParameterConfig



class MyDataset(Dataset):
    def __init__(self, data: list[str]):
        super().__init__()
        self.data = data
    def __len__(self):
        return len(self.data)
    def __getitem__(self, index):
        return self.data[index]
